# -*- coding: utf-8 -*-
# @Time    : 2024/7/24 10:54
# @Author  : yujiahao
# @File    : 57_fastapi_header.py
# @description:响应头


from fastapi import FastAPI, Response
from starlette.responses import JSONResponse

app = FastAPI()


# todo 使用 Response 参数
# 可以在你的路径操作函数中声明一个Response类型的参数（就像你可以为cookies做的那样）。
# 然后你可以在这个临时响应对象中设置头部。
@app.get("/headers-and-object/")
def get_headers(response: Response):
    """
    然后你可以像平常一样返回任何你需要的对象（例如一个 dict 或者一个数据库模型）。
    如果你声明了一个 response_model，它仍然会被用来过滤和转换你返回的对象。

    FastAPI 将使用这个临时响应来提取头部（也包括 cookies 和状态码），
    并将它们放入包含你返回的值的最终响应中，该响应由任何 response_model 过滤。

    你也可以在依赖项中声明 Response 参数，并在其中设置头部（和 cookies）。
    """

    response.headers["X-Cat-Dog"] = "alone in the world"
    return {"message": "Hello World"}


# todo 直接返回 Response

# 你也可以在直接返回Response时添加头部。
# 按照直接返回响应中所述创建响应，并将头部作为附加参数传递：
@app.get("/headers/")
def get_headers():
    content = {"message": "Hello World"}
    headers = {"X-Cat-Dog": "alone in the world", "Content-Language": "en-US"}
    return JSONResponse(content=content, headers=headers)

# todo 当然了也可以自定义头部
# 需要将它们添加到你的CORS配置中
